from django.contrib import admin

from TestInterface.models import TestInterfaceModel, TestInterfaceCaseModel


# Register your models here.

@admin.register(TestInterfaceModel)
class TestInterfaceModelAdmin(admin.ModelAdmin):
    list_display = ['id', 'project', 'name', 'url', 'method', 'type']


@admin.register(TestInterfaceCaseModel)
class TestInterfaceCaseModelAdmin(admin.ModelAdmin):
    list_display = ['id', 'title', 'interface', 'headers', 'request', 'file', 'setup_script', 'teardown_script']
